from setuptools import setup, find_packages

setup(
    name="qtorch",
    version="1.0.0",
    packages=find_packages(where="src"),
    package_dir={"": "src"},
    install_requires=[
        "numpy>=1.20.0",
        "pandas>=1.3.0",
        "matplotlib>=3.4.0",
        "seaborn>=0.11.0",
        "pyyaml>=6.0",
        "akshare>=1.0.0",  # 用于获取股票、ETF、外汇等市场数据
        "scipy>=1.7.0",
        "pyarrow>=6.0.0",  # 用于parquet文件支持
        "tqdm>=4.62.0",    # 进度条
        "torch>=2.0.0",    # PyTorch核心库
        "scikit-learn>=1.0.0",  # 用于机器学习算法和评估
        "pykalman"       # 卡尔曼滤波
    ],
    python_requires=">=3.6",
)